package com.lagou.base;

import com.alibaba.fastjson.JSON;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
import java.io.IOException;
import java.lang.reflect.Method;
import java.util.Map;

/**
 * @Author wsl
 * @Date 2021/10/6 23:45
 * @Project_Name maven
 */
public class BaseServlet extends HttpServlet {
    /**
     * doGET方法作为一个调度器去使用，根据请求功能的不同，去调用对应的方法
     *   规定必须传一个参数
     *     methodName=功能名
     * @param req
     * @param resp
     * @throws ServletException
     * @throws IOException
     */
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.获取参数，要访问的方法名
        //String methodName = req.getParameter("methodName");
        String methodName = null;

        //1.获取POST请求的，Content-Type类型
        String contentType = req.getHeader("Content-Type");

        //2.判断传递的数据是否为JSON格式
        if ("application/json;charset=utf-8".equals(contentType)){
            //是JSON格式，调用getPostJSON方法
            String postJSON = getPostJSON(req);

            //将JSON格式的字符串 转换为 map集合
            Map<String,Object> map = JSON.parseObject(postJSON, Map.class);

            //从map集合中 获取 methodName
            methodName = (String) map.get("methodName");

            //将获取到的数据，保存到request域对象中
            req.setAttribute("map",map);

        } else {
            methodName = req.getParameter("methodName");
        }

        //2.判断，执行对应的方法
        if (methodName != null){
            //通过反射优化代码，提高代码的可维护性
            try {
                //1.获取字节码文件对象， this = TestServlet
                Class c = this.getClass();

                //2.根据传入的方法名，获取对应的方法对象
                Method method = c.getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class);

                //3.调用method对象的 invoke方法，执行对应的功能
                method.invoke(this,req,resp);

            } catch (Exception e) {
                e.printStackTrace();
                System.out.println("请求的功能不存在！");
            }
        }

        /*if (null != methodName && methodName.equals("addCourse")){
            addCourse(req, resp);
        } else if (null != methodName && methodName.equals("findByName")){
            findByName(req, resp);
        } else if (null != methodName && methodName.equals("findByStatus")){
            findByStatus(req, resp);
        } else {
            System.out.println("没有此功能对应的方法...");
        }*/
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        doGet(req, resp);
    }

    /**
     * POST请求格式是：Content-Type : application/json ; charset=utf-8
     *   使用该方法进行读取
     */
    public String getPostJSON(HttpServletRequest request){

        try {
            //1.从request中 ，获取缓冲输入流对象
            BufferedReader reader = request.getReader();

            //2.创建StringBuffer, 保存读取出的数据
            StringBuffer sb = new StringBuffer();

            //3.循环读取
            String line = null;
            while ( (line = reader.readLine()) != null){
                 //将每次读取到的数据，追加到StringBuffer中去
                sb.append(line);
            }

            //4.返回结果
            return sb.toString();

        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }

}
